#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2020/9/28 9:33
# @Author  : 朱宽
# @File    : manageCookies.py
# @Software: Win10、Python3.8.5 and Pycharm
'''
处理cookie，11.12
'''


class manageCookies():
    '''
    定义了：
    1、删除所有cookie
    2、获取当前所有cookie
    3、获取当前 名称 为 key的cookie信息
    4、添加cookie，cookie_dict参数为字典
    '''

    def __init__(self, driver):
        self.driver = driver

    def delete_current_cookie(self):
        '''
        1、删除所有cookie
        :return:
        '''
        self.driver.delete_all_cookies()

    def get_current_cookies(self):
        '''
        2、获取当前所有cookie
        :return:
        '''
        current_cookie = self.driver.get_cookies()
        return current_cookie

    def get_current_cookie_value(self, key):
        '''
        3、获取当前 名称 为 key的cookie信息
        :param key:
        :return:
        '''
        key_cookie = self.driver.get_cookie(key)
        return key_cookie

    def add_key_value_to_cookie(self, cookie_dict):
        '''
        4、添加cookie，cookie_dict参数为字典
        :param cookie_dict:
        :return:
        '''
        self.driver.add_cookie(cookie_dict)
